// package cn.com.nes.mybatis;
//
// import com.baomidou.mybatisplus.exceptions.MybatisPlusException;
// import com.baomidou.mybatisplus.generator.AutoGenerator;
// import com.baomidou.mybatisplus.generator.InjectionConfig;
// import com.baomidou.mybatisplus.generator.config.*;
// import com.baomidou.mybatisplus.generator.config.po.TableInfo;
// import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
// import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine;
// import org.apache.commons.lang3.StringUtils;
//
// import java.util.ArrayList;
// import java.util.List;
// import java.util.Scanner;
//
// public class MyAutoGenerator {
//     /**
//      * <p>
//      * 读取控制台内容
//      * </p>
//      */
//     public static String scanner(String tip) {
//         Scanner scanner = new Scanner(System.in);
//         System.out.println("请输入" + tip + "：");
//         if (scanner.hasNext()) {
//             String ipt = scanner.next();
//             if (StringUtils.isNotBlank(ipt)) {
//                 return ipt;
//             }
//         }
//         throw new MybatisPlusException("请输入正确的" + tip + "！");
//     }
//
//     public static void main(String[] args) {
//         // 配置数据库连接信息
// //        final String dbname = "nes_agent";
//         final String dbname = "nes_ope_prod";
//         final String url = "jdbc:mysql://rm-2ze3340731300n995no.mysql.rds.aliyuncs.com:3306/" + dbname + "?characterEncoding=utf-8";
//         final String driverName = "com.mysql.jdbc.Driver";
//         final String root = "dbadmin";
//         final String password = "db%NES2021!";
//
//         // 代码生成器
//         AutoGenerator mpg = new AutoGenerator();
//
//         // 全局配置
//         GlobalConfig gc = new GlobalConfig();
//         final String projectPath = System.getProperty("user.dir");
//         gc.setOutputDir(projectPath + "/src/main/java");
//         gc.setAuthor("nes");
//         gc.setOpen(false);
//         // gc.setSwagger2(true); 实体属性 Swagger2 注解
//         mpg.setGlobalConfig(gc);
//
//         gc.setBaseResultMap(true);
//         gc.setBaseColumnList(true);
//         gc.setEnableCache(false);
//
//         // 数据源配置
//         DataSourceConfig dsc = new DataSourceConfig();
//         dsc.setUrl(url);
//         dsc.setDriverName(driverName);
//         dsc.setUsername(root);
//         dsc.setPassword(password);
//
//         mpg.setDataSource(dsc);
//
//         // 包配置
//         final PackageConfig pc = new PackageConfig();
//         pc.setModuleName(scanner("模块名"));
//         pc.setParent("cn.com.nes.mybatis");
//         mpg.setPackageInfo(pc);
//
//         // 自定义配置
//         InjectionConfig cfg = new InjectionConfig() {
//             @Override
//             public void initMap() {
//                 // to do nothing
//             }
//         };
//
//         // 如果模板引擎是 freemarker
// //        String templatePath = "/templates/mapper.xml.ftl";
//         // 如果模板引擎是 velocity
//         String templatePath = "/templates/mapper.xml.vm";
//
//         // 自定义输出配置
//         List<FileOutConfig> focList = new ArrayList<>();
//         // 自定义配置会被优先输出
//         focList.add(new FileOutConfig(templatePath) {
//             @Override
//             public String outputFile(TableInfo tableInfo) {
//                 // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
//                 return projectPath + "/src/main/resources/mapper/" + pc.getModuleName()
//                         + "/" + tableInfo.getEntityName() + "Mapper" + ".xml";
//             }
//         });
//         /*
//         cfg.setFileCreate(new IFileCreate() {
//             @Override
//             public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {
//                 // 判断自定义文件夹是否需要创建
//                 checkDir("调用默认方法创建的目录，自定义目录用");
//                 if (fileType == FileType.MAPPER) {
//                     // 已经生成 mapper 文件判断存在，不想重新生成返回 false
//                     return !new File(filePath).exists();
//                 }
//                 // 允许生成模板文件
//                 return true;
//             }
//         });
//         */
//         cfg.setFileOutConfigList(focList);
//         mpg.setCfg(cfg);
//
//         // 配置模板
//         TemplateConfig templateConfig = new TemplateConfig();
//
//         // 配置自定义输出模板
//         //指定自定义模板路径，注意不要带上.ftl/.vm, 会根据使用的模板引擎自动识别
//         // templateConfig.setEntity("templates/entity2.java");
//         templateConfig.setService(null);
//         templateConfig.setServiceImpl(null);
//         templateConfig.setController(null);
//         templateConfig.setXml(null);
//         mpg.setTemplate(templateConfig);
//
//         // 策略配置
//         StrategyConfig strategy = new StrategyConfig();
//         strategy.setNaming(NamingStrategy.underline_to_camel);
//         strategy.setColumnNaming(NamingStrategy.underline_to_camel);
// //        strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
//         strategy.setEntityLombokModel(false);
//         strategy.setEntityColumnConstant(true);
// //        strategy.setRestControllerStyle(true);
//         // 公共父类
// //        strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!");
//         // 写于父类中的公共字段
// //        strategy.setSuperEntityColumns("id");
//         strategy.setInclude(scanner("表名，多个英文逗号分割").split(","));
// //        strategy.setControllerMappingHyphenStyle(true);
// //        strategy.setTablePrefix(pc.getModuleName() + "_");
//         mpg.setStrategy(strategy);
//         mpg.setTemplateEngine(new VelocityTemplateEngine());
//         mpg.execute();
//     }
// }
